package com.headwind.logistics.dispatchNetwork.util;

import com.headwind.logistics.dispatchNetwork.dto.SiteArerTreeDto;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author DELL
 * @Date 2020/7/24 14:54
 * @Version 1.0
 * @DESCRIPTION 填写类作用描述
 */
public class SiteTreeUtil {

    private List<SiteArerTreeDto> menuList = new ArrayList<>();

    /*public SiteTreeUtil(List<SiteArerTreeDto> menuList) {

        this.menuList=menuList;

    }


    //建立树形结构

    public List<SiteArerTreeDto> builTree(){


        List<SiteArerTreeDto> moduleMenus =new  ArrayList<SiteArerTreeDto>();

        for(SiteArerTreeDto module : getRootModule()) {


            module=buildChilTree(module);

            moduleMenus.add(module);

        }

        return moduleMenus;

    }


    //递归，建立子树形结构

    private SiteArerTreeDto buildChilTree(SiteArerTreeDto module){

        List<SiteArerTreeDto> chilMenus =new  ArrayList<SiteArerTreeDto>();


        for(SiteArerTreeDto menuModule : menuList) {

            if(menuModule.getModModuleId().equals(module.getModuleId())) {

                chilMenus.add(buildChilTree(menuModule));

            }

        }

        module.setModules(chilMenus);

        return module;

    }


    //获取根节点

    private List<SiteArerTreeDto> getRootModule() {

        List<SiteArerTreeDto> modules =new  ArrayList<SiteArerTreeDto>();

        for(SiteArerTreeDto module : menuList) {

            if(module.getModModuleId().equals("0")) {

                modules.add(module);

            }

        }

        return modules;

    }*/
}
